import {Annotation, StateGraph} from "@langchain/langgraph";
import {qwen} from "../models";
import {ChatPromptTemplate} from "@langchain/core/prompts";
import {z} from 'zod'

const question = '领绣集团2022年的市场表现如何？'

console.log("提问：----",question)

const GraphState = Annotation.Root({

})

const template2 = ChatPromptTemplate.fromTemplate(`
    
    给定一个问题，判断这个问题是定量还是定性类型，只需要返回"定量"或者"定性"
    question: {question}

`)

const model2 = template2.pipe(qwen);
const rep2 = await model2.invoke({
    question: question
})
console.log("问题类型：-------", rep2.content)


const questions = z.object({
    questions: z.array(z.string()).describe('问题列表')
})

const template = ChatPromptTemplate.fromTemplate(`
    
    你是一个数据分析助手，基于给定的背景知识和维度指标，对问题进行理解，并量化返回成单个或者多个问题，方便分析和获取数据。
    背景如下：
    '''
    浙江联翔智能家居股份有限公司，成立于2004年，是首批从事刺绣工艺墙布研发与生产的企业之一，现已是一家集设计、研发、生产、销售及运营为一体的综合性软装企业。
    当前场景包含指标：销售金额、销售米数
    维度：产品名称、产品颜色、品牌、区域经理、版本、客户
    '''
    question: {question}

`)


const model = template.pipe(qwen.withStructuredOutput(questions))

const rep1 = await model.invoke({
    question: question
})

console.log('问题列表：', rep1)
//
//
// const template3 = ChatPromptTemplate.fromTemplate(`
//
//     你是一个数据分析专家，基于给定的背景知识和维度指标，对给定的定性问题生成研究分析目录。
//     背景如下：
//     '''
//     浙江联翔智能家居股份有限公司，成立于2004年，是首批从事刺绣工艺墙布研发与生产的企业之一，现已是一家集设计、研发、生产、销售及运营为一体的综合性软装企业。
//     当前场景包含指标：销售金额、销售米数
//     维度：产品名称、产品颜色、品牌、区域经理、版本、客户
//     '''
//     question: {question}
//
// `)
//
//
// const model3 = template3.pipe(qwen)
//
// const rep3 = await model3.invoke({
//     question: question
// });
//
// console.log('目录结构：-----', rep3.content)
//
//



// const template2 = ChatPromptTemplate.fromTemplate(`
//     你是一个数据分析专家，给定一个定性问题，理解这个问题，把定性问题转成多个定量问题，帮助我们更有效的收集和分析数据。
//     question: {question}
// `)
// const model2 = template2.pipe(qwen)
//
// console.log(await model2.invoke({
//     question: '海宁水务去年营销情况怎么样'
// }))
